import sys
import os
sys.path.append('src')
from data_augmentation.augment import data_augmentation_main


data_augmentation_main('/disk1/wyf/Project/zz_static_project/dataset/C/testcases/CWE121_Stack_Based_Buffer_Overflow/s01',
                       '/disk1/wyf/Project/zz_static_project/dataset/secure_dataset',
                       '/disk1/wyf/Project/zz_static_project/dataset/augmented_data')